#include "UdpServer.hpp"
#include <vector>
#include <memory>



void Usage(std::string proc)
{
    std::cout << "\n\rUsage: " << proc << " port[1024+]\n" << std::endl;
}

// UDP 服务端要对 用户发送过来的数据执行的功能
std::string Hander(const std::string& s)
{
    std::string ret = "UdpServer get a message: " + s;
    return ret;
}

// Udp 指令处理
bool CheckCommand(const std::string &cmd)
{
    std::vector<std::string> key_word = {"rm", "mv", "cp", "kill"};
    for(auto& key : key_word)
    {
        if(key == cmd) return false;
    }
    return true;
}
std::string ExcuteCommand(const std::string& s)
{
    std::cout << "get a message: " << s << std::endl;
    // 检测 用户 输入的指令是否合法
    if(!CheckCommand(s))
        return "This command is prohibited";
    FILE* fp = popen(s.c_str(), "r");   // popen: fork&&pipe
    if(fp == nullptr){
        perror("popen");
        return "error";
    }

    std::string result;
    char buffer[SIZE*4];
    while (true)
    {
        // 从管道文件中获取信息
        char *ok = fgets(buffer, sizeof(buffer), fp);
        if(ok == nullptr) break;    // 信息获取完毕
        result += buffer;
    }
    pclose(fp);
    return result;
}

// ./udpserver port
int main(int argc, char *argv[])
{
    if(argc != 2){
        Usage(argv[0]);
        exit(0);
    }
    uint16_t port = atoi(argv[1]);

    // 开始处理
    std::unique_ptr<UdpServer> svr(new UdpServer(port));
    svr->Init();
    // svr->Run(ExcuteCommand);
    svr->Run();
    return 0;
}